#==============================================================================
#  @@-COPYRIGHT-START-@@
#
#  Copyright (c) 2022, Qualcomm Innovation Center, Inc. All rights reserved.
#
#  Redistribution and use in source and binary forms, with or without
#  modification, are permitted provided that the following conditions are met:
#
#  1. Redistributions of source code must retain the above copyright notice,
#     this list of conditions and the following disclaimer.
#
#  2. Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions and the following disclaimer in the documentation
#     and/or other materials provided with the distribution.
#
#  3. Neither the name of the copyright holder nor the names of its contributors
#     may be used to endorse or promote products derived from this software
#     without specific prior written permission.
#
#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
#  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
#  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
#  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
#  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
#  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
#  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
#  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
#  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
#  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
#  POSSIBILITY OF SUCH DAMAGE.
#
#  SPDX-License-Identifier: BSD-3-Clause
#
#  @@-COPYRIGHT-END-@@
#==============================================================================

add_subdirectory(src/python)

if (ENABLE_TESTS)
    add_subdirectory(test)
endif()

add_library(OnnxCppOps SHARED
        src/QcQuantizeOp.h
        src/AimetOpUtils.h
        src/AimetOpUtils.cpp
        src/RegisterCustomOps.h
        src/RegisterCustomOps.cpp
        src/QcQuantizeInfo.cpp
        src/QcQuantizeInfo.h
        src/QcQuantizeOp.cpp
        src/OnnxOpUtils.h
)

target_include_directories(OnnxCppOps PRIVATE
        ${EIGEN3_INCLUDE_DIR}
        ${ONNXRUNTIME_INCLUDE_DIR}
        ${CMAKE_CURRENT_SOURCE_DIR}/../common/include
        )

target_link_libraries(OnnxCppOps PUBLIC
        MoDlQuantization
        )

if (ENABLE_CUDA)
    target_compile_options(OnnxCppOps PRIVATE
            -DONNX_CUDA=1
            )

    target_link_libraries(OnnxCppOps PUBLIC
            MoDlQuantizationCuda
            )
endif (ENABLE_CUDA)

if (SKBUILD)
    set_target_properties(OnnxCppOps PROPERTIES
        OUTPUT_NAME "aimet_onnxrt_ops"
        )

    install(TARGETS OnnxCppOps
        DESTINATION aimet_common
        )

else (SKBUILD)
    set_target_properties(OnnxCppOps PROPERTIES
        OUTPUT_NAME "aimet_onnxrt_ops"
        LIBRARY_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/artifacts/aimet_common"
        )

install(TARGETS OnnxCppOps
        DESTINATION ${CMAKE_INSTALL_LIBDIR}/python/aimet_common
        )

endif (SKBUILD)

# ----

Python3_add_library(quant_info SHARED
        src/QcQuantizeInfo.h
        src/QcQuantizeInfo.cpp)

    target_link_libraries(quant_info PUBLIC
            MoDlQuantization
            )

if (ENABLE_CUDA)
    target_link_libraries(quant_info PUBLIC
            MoDlQuantizationCuda
            )
endif (ENABLE_CUDA)

if (SKBUILD)
    set_target_properties(quant_info
        PROPERTIES
        OUTPUT_NAME "quant_info"
        )

    install(TARGETS quant_info
            DESTINATION aimet_common
            )

else (SKBUILD)

set_target_properties(quant_info
        PROPERTIES
        OUTPUT_NAME "quant_info"
        LIBRARY_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/artifacts/aimet_common/"
        )

install(TARGETS quant_info
        DESTINATION ${CMAKE_INSTALL_LIBDIR}/python/aimet_common
        )

# ----

add_custom_target(whl_prep_cp_onnx
    COMMAND ${CMAKE_COMMAND} -E make_directory ${WHL_PREP_AIMET_ONNX_DIR}
    COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:OnnxCppOps> ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:OnnxCppOps>
    COMMAND ${PATCHELF_EXE} --set-rpath '$$ORIGIN' ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:OnnxCppOps>
    COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:quant_info> ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:quant_info>
    COMMAND ${PATCHELF_EXE} --set-rpath '$$ORIGIN' ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:quant_info>
    DEPENDS OnnxCppOps quant_info
)
add_custom_target(whl_prep_ln_onnx
    COMMAND ${CMAKE_COMMAND} -E make_directory ${WHL_PREP_AIMET_ONNX_DIR}
    COMMAND ${CMAKE_COMMAND} -E create_symlink $<TARGET_FILE:OnnxCppOps> ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:OnnxCppOps>
    COMMAND ${CMAKE_COMMAND} -E create_symlink $<TARGET_FILE:quant_info> ${WHL_PREP_AIMET_ONNX_DIR}/$<TARGET_FILE_NAME:quant_info>
    DEPENDS OnnxCppOps quant_info
)

add_dependencies(whl_prep_cp whl_prep_cp_onnx)
add_dependencies(whl_prep_ln whl_prep_ln_onnx)

endif (SKBUILD)

if (${CMAKE_SYSTEM_NAME} STREQUAL "Windows")
    set_target_properties(quant_info
        PROPERTIES
        SUFFIX ".${Python3_SOABI}.pyd"
        )
else()
    set_target_properties(quant_info
        PROPERTIES
        SUFFIX ".${Python3_SOABI}.so"
        )
endif()
